library(haven)

# Load in Data set
impdat20_disab <- read_spss("./data/Disability_IAT.public.2020.sav")
# Subset to USA
impdat20_disab <- subset(impdat20_disab, countryres_num == 1)

# Meta Info ---------------------------------------------------------------
impdat20_disab$date_mdy <- format(impdat20_disab$date, "%Y-%m-%d")
impdat20_disab$date_mdy <- as.Date(impdat20_disab$date_mdy)
impdat20_disab$week <- format(as.Date(impdat20_disab$date), "%W")

# Demographics ---------------------------------------------------------------
# Age
impdat20_disab$age[which(impdat20_disab$birthyear > 1900)] <- impdat20_disab$year[which(impdat20_disab$birthyear > 1900)] - impdat20_disab$birthyear[which(impdat20_disab$birthyear > 1900)]
impdat20_disab$age_sc <- (impdat20_disab$age-min(impdat20_disab$age, na.rm = T))/(max(impdat20_disab$age, na.rm = T)-min(impdat20_disab$age, na.rm = T))

impdat20_disab$age_cat4 <- NA
impdat20_disab$age_cat4[which(impdat20_disab$age >= 18 & impdat20_disab$age <= 29)] <- "18-29"
impdat20_disab$age_cat4[which(impdat20_disab$age >= 30 & impdat20_disab$age <= 44)] <- "30-44"
impdat20_disab$age_cat4[which(impdat20_disab$age >= 45 & impdat20_disab$age <= 64)] <- "45-64"
impdat20_disab$age_cat4[which(impdat20_disab$age >= 65)] <- "65+"
impdat20_disab$age_cat4 <- as.factor(impdat20_disab$age_cat4)

# Sex
impdat20_disab$sex <- NA
impdat20_disab$sex[which(impdat20_disab$birthSex == 2)] <- "f"
impdat20_disab$sex[which(impdat20_disab$birthSex == 1)] <- "m"

# Race and Ethnicity
impdat20_disab$race <- factor(impdat20_disab$raceomb_002, 
                           levels = c(1:max(impdat20_disab$raceomb_002, na.rm = T)),
                           labels = c("American Indian/Alaskan Native",
                                      "East Asian",
                                      "South Asian",
                                      "Native Hawaiian/Pacific Islander",
                                      "Black",
                                      "White",
                                      "Other", 
                                      "Multiracial"))

impdat20_disab$hisp <- NA
impdat20_disab$hisp[which(impdat20_disab$ethnicityomb == 1)] <- 1
impdat20_disab$hisp[which(impdat20_disab$ethnicityomb == 2)] <- 0 # drops unknowns

impdat20_disab$race5 <- NA
impdat20_disab$race5[which(impdat20_disab$race %in% c("East Asian", "South Asian", "Native Hawaiian/Pacific Islander") & impdat20_disab$hisp == 0)] <- "Asian/Pacific Islander"
impdat20_disab$race5[which(impdat20_disab$race == "Black" & impdat20_disab$hisp == 0)] <- "Black"
impdat20_disab$race5[which(impdat20_disab$race == "White" & impdat20_disab$hisp == 0)] <- "White"
impdat20_disab$race5[which(impdat20_disab$race %in% c("Other", "Multiracial") & impdat20_disab$hisp == 0)] <- "Other"
impdat20_disab$race5[which(impdat20_disab$hisp == 1)] <- "Hispanic"
impdat20_disab$race5 <- as.factor(impdat20_disab$race5)
impdat20_disab$race5 <- factor(impdat20_disab$race5, levels(impdat20_disab$race5)[c(1:3,5,4)])

impdat20_disab$race3 <- NA
impdat20_disab$race3[which(impdat20_disab$race5 == "White")] <- "White"
impdat20_disab$race3[which(impdat20_disab$race5 == "Black")] <- "Black"
impdat20_disab$race3[which(impdat20_disab$race5 != "Black" & impdat20_disab$race5 != "White")] <- "Other"

# Ideology
impdat20_disab$ideo3 <- NA
impdat20_disab$ideo3[which(impdat20_disab$politicalid_7 < 4)] <- 1
impdat20_disab$ideo3[which(impdat20_disab$politicalid_7 == 4)] <- 0
impdat20_disab$ideo3[which(impdat20_disab$politicalid_7 > 4)] <- -1
impdat20_disab$ideo3_lab <- factor(impdat20_disab$ideo3,
                                labels = c("liberal", "moderate", "conservative"))

impdat20_disab$ideo_lib <- ifelse(impdat20_disab$ideo3 == -1, 1, 0)
impdat20_disab$ideo_mod <- ifelse(impdat20_disab$ideo3 == 0, 1, 0)
impdat20_disab$ideo_con <- ifelse(impdat20_disab$ideo3 == 1, 1, 0)

impdat20_disab$ideo7 <- impdat20_disab$politicalid_7*-1 + 7
impdat20_disab$ideo7_sc <- impdat20_disab$ideo7/6

# Region
impdat20_disab$south <- NA
impdat20_disab$south <- ifelse(impdat20_disab$STATE %in% c("VA", "NC", "SC", "GA", "FL", "MS", "AL", "TN", "AR", "LA", "TX"),
                            1, 0)
impdat20_disab$south[impdat20_disab$STATE == ""] <- NA

impdat20_disab$census_region <- NA
impdat20_disab$census_region[which(impdat20_disab$STATE %in% c("DE", "MD", "DC", "WV", "KY", "OK",
                                                         "VA", "NC", "SC", "GA", "FL", "MS", "AL", 
                                                         "TN", "AR", "LA", "TX"))] <- "South"
impdat20_disab$census_region[which(impdat20_disab$STATE %in% c("CA", "OR", "WA", "MT", "WY",
                                                         "ID", "CO", "UT", "AZ", "NM",
                                                         "NV", "AK", "HI"))] <- "West"
impdat20_disab$census_region[which(impdat20_disab$STATE %in% c("ND", "SD", "MN", "WI",
                                                         "IA", "NE", "KS", "MO",
                                                         "IL", "IN", "MI", "OH"))] <- "Midwest"
impdat20_disab$census_region[which(impdat20_disab$STATE %in% c("ME", "VT", "NH", "MA", "RI",
                                                         "CT", "NY", "NJ", "PA"))] <- "Northeast"
impdat20_disab$census_region[impdat20_disab$STATE == ""] <- NA
impdat20_disab$census_region <- as.factor(impdat20_disab$census_region)

#Other vars: CountyNo, MSANo, MSAName

# Education
impdat20_disab$edu_cat <- NA
impdat20_disab$edu_cat[which(impdat20_disab$edu <= 3)] <- 1 # Less than HS
impdat20_disab$edu_cat[which(impdat20_disab$edu == 4)] <- 2 # HS 
impdat20_disab$edu_cat[which(impdat20_disab$edu == 5 | impdat20_disab$edu == 6)] <- 3 # Some college or associates
impdat20_disab$edu_cat[which(impdat20_disab$edu == 7)] <- 4 # College
impdat20_disab$edu_cat[which(impdat20_disab$edu >= 8)] <- 5 # Adv degree or schooling
impdat20_disab$col <- ifelse(impdat20_disab$edu_cat >= 4, 1, 0)

impdat20_disab$edu_cat4 <- NA
impdat20_disab$edu_cat4[which(impdat20_disab$edu_cat <= 2)] <- "HS or Less"
impdat20_disab$edu_cat4[which(impdat20_disab$edu_cat == 3)] <- "Some College"
impdat20_disab$edu_cat4[which(impdat20_disab$edu_cat == 4)] <- "College"
impdat20_disab$edu_cat4[which(impdat20_disab$edu_cat == 5)] <- "Post-Grad"
impdat20_disab$edu_cat4 <- factor(impdat20_disab$edu_cat4,
                               levels = c("HS or Less",
                                          "Some College",
                                          "College",
                                          "Post-Grad"))

# Reason for Visiting
# impdat20_disab$broughtwebsite
impdat20_disab$broughtwebsite2 <- NA
impdat20_disab$broughtwebsite2[which(impdat20_disab$broughtwebsite %in% c("Assignment for school", "Assignment for work"))] <- "Assignment school/work"
impdat20_disab$broughtwebsite2[which(impdat20_disab$broughtwebsite %in% c("Mention in a news story (any medium)"))] <- "News Mention"
impdat20_disab$broughtwebsite2[which(impdat20_disab$broughtwebsite %in% c("My Internet search for this topic or a related topic", "Mention or link at a non-news Internet site"))] <- "Internet Mention/Search"
impdat20_disab$broughtwebsite2[which(impdat20_disab$broughtwebsite %in% "Recommendation of a friend or co-worker")] <- "Peer Recommendation"
impdat20_disab$broughtwebsite2[which(impdat20_disab$broughtwebsite %in% c("Other", "null"))] <- "Other/null"

impdat20_disab$broughtwebsite2 <- as.factor(impdat20_disab$broughtwebsite2)


impdat20_disab$broughtwebsite4 <- NA
impdat20_disab$broughtwebsite4[which(impdat20_disab$broughtwebsite %in% c("Assignment for school", "Assignment for work"))] <- "Assignment school/work"
impdat20_disab$broughtwebsite4[which(impdat20_disab$broughtwebsite %in% c("My Internet search for this topic or a related topic", "Mention or link at a non-news Internet site", "News Mention"))] <- "News/Internet Mention/Search"
impdat20_disab$broughtwebsite4[which(impdat20_disab$broughtwebsite %in% "Recommendation of a friend or co-worker")] <- "Peer Recommendation"
impdat20_disab$broughtwebsite4[which(impdat20_disab$broughtwebsite %in% c("Other", "null"))] <- "Other/null"

impdat20_disab$broughtwebsite4 <- as.factor(impdat20_disab$broughtwebsite4)

# Interview Date Indicators -----------------------------------------------
impdat20_disab$post_gf <- ifelse(impdat20_disab$date_mdy > "2020-05-25", 1, 0)
impdat20_disab$post_gf_1month <- ifelse(impdat20_disab$date_mdy > "2020-05-25" & impdat20_disab$date_mdy <= "2020-06-26", 
                                     1, 0)
impdat20_disab$post_gf_2month <- ifelse(impdat20_disab$date_mdy > "2020-06-26" & impdat20_disab$date_mdy <= "2020-07-26", 
                                     1, 0)
impdat20_disab$post_gf_3month <- ifelse(impdat20_disab$date_mdy > "2020-07-26" & impdat20_disab$date_mdy <= "2020-08-26", 
                                     1, 0)
